package com.ruoyi.ade.llm.model;

import java.util.List;

/**
 * rerank 请求体
 */
public class RerankRequest {

    public static final String model_rerank_bce_reranker_base= "bce-reranker-base";

    private String model;
    private String query;
    private List<String> documents;
    private int top_n;

    public RerankRequest(String model, String query, List<String> documents, int top_n){
        this.model = model;
        this.query = query;
        this.documents = documents;
        this.top_n = top_n;
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public String getQuery() {
        return query;
    }

    public void setQuery(String query) {
        this.query = query;
    }

    public List<String> getDocuments() {
        return documents;
    }

    public void setDocuments(List<String> documents) {
        this.documents = documents;
    }

    public int getTop_n() {
        return top_n;
    }

    public void setTop_n(int top_n) {
        this.top_n = top_n;
    }
}
